|
1
|
2 |
|
import {Rat} from './Rat' |
|
2
|
2 |
|
import Symbolizer from './Symbolizer' |
|
3
|
|
|
|
|
4
|
|
|
interface Coefficents<Rat> { |
|
5
|
|
|
[Key: string]: Rat; |
|
6
|
|
|
} |
|
7
|
|
|
|
|
8
|
|
|
/** |
|
9
|
|
|
* @class Rational polynumber |
|
10
|
|
|
* @name Polyrat |
|
11
|
|
|
*/ |
|
12
|
2 |
|
export class Polyrat { |
|
13
|
|
|
|
|
14
|
|
|
// coefficent values are indexed with their the exponents in each dimension, comma-separated, as the key |
|
15
|
7 |
|
coefficents: Coefficents<Rat> = {} |
|
16
|
|
|
|
|
17
|
|
|
// the dimension is how many params there are, defined by the length of the exponent keys |
|
18
|
7 |
|
dimension = 0 |
|
19
|
|
|
|
|
20
|
|
|
// unique symbols for each dimension |
|
21
|
7 |
|
symbols = '' |
|
22
|
|
|
|
|
23
|
|
|
/** |
|
24
|
|
|
* Initialize a rational polynumber. |
|
25
|
|
|
*/ |
|
26
|
|
|
constructor(coefficents?: Coefficents<Rat>) { |
|
27
|
7 |
|
if (coefficents) this.coefficents = coefficents |
|
28
|
7 |
|
if (Object.keys(this.coefficents).length) { |
|
29
|
4 |
|
this.dimension = Object.keys(this.coefficents)[0].split(',').length |
|
30
|
|
|
} |
|
31
|
7 |
|
const sg = (new Symbolizer('xyzw')).generator() |
|
32
|
7 |
|
for (let i=0; i<this.dimension; i++) { |
|
33
|
5 |
|
this.symbols += sg.next().value |
|
34
|
|
|
} |
|
35
|
|
|
} |
|
36
|
|
|
|
|
37
|
|
|
/** |
|
38
|
|
|
* Evaluate the result given the parameters for each dimension. |
|
39
|
|
|
*/ |
|
40
|
|
|
evaluate(parameters: Rat[]): Rat { |
|
41
|
1 |
|
let result: Rat = new Rat() |
|
42
|
1 |
|
for (const [exponents, coefficent] of Object.entries(this.coefficents)) { |
|
43
|
1 |
|
let value: Rat = coefficent |
|
44
|
1 |
|
const dimensions = exponents.split(',') |
|
45
|
1 |
|
for (let i=0; i<dimensions.length; i++) { |
|
46
|
1 |
|
value = value.mul(parameters[i].pow(new Rat(parseInt(dimensions[i], 10)))) |
|
47
|
|
|
} |
|
48
|
1 |
|
result = result.add(value) |
|
49
|
|
|
} |
|
50
|
1 |
|
return result |
|
51
|
|
|
} |
|
52
|
|
|
|
|
53
|
|
|
/** |
|
54
|
|
|
* The text representation. |
|
55
|
|
|
*/ |
|
56
|
|
|
toString(): string { |
|
57
|
|
|
// return JSON.stringify(this.coefficents) |
|
58
|
3 |
|
const r = [] |
|
59
|
3 |
|
for (const [exponents, coefficent] of Object.entries(this.coefficents)) { |
|
60
|
2 |
|
r.push(`'${exponents}': ${coefficent.toString()}'`) |
|
61
|
|
|
} |
|
62
|
3 |
|
return `[${r.join(',')}]` |
|
63
|
|
|
} |
|
64
|
|
|
|
|
65
|
|
|
/** |
|
66
|
|
|
* The "calc" code for evaluating the value. |
|
67
|
|
|
*/ |
|
68
|
|
|
toCalcFormula(): string { |
|
69
|
1 |
|
const r: string[] = [] |
|
70
|
1 |
|
for (const [exponents, coefficent] of Object.entries(this.coefficents)) { |
|
71
|
5 |
|
const t: string[] = [] |
|
72
|
5 |
|
const f = coefficent.toString() |
|
73
|
5 |
|
if (f !== '1') t.push(f) |
|
74
|
5 |
|
const dimensions = exponents.split(',') |
|
75
|
5 |
|
for (let i=0; i<dimensions.length; i++) { |
|
76
|
10 |
|
if (dimensions[i] !== '0') { |
|
77
|
6 |
|
t.push(`${this.symbols[i]}^${parseInt(dimensions[i], 10)}`) |
|
78
|
|
|
} |
|
79
|
|
|
} |
|
80
|
5 |
|
if (t) r.push(t.join('*')) |
|
81
|
|
|
} |
|
82
|
1 |
|
return r.join(' + ') |
|
83
|
|
|
} |
|
84
|
|
|
|
|
85
|
|
|
/** |
|
86
|
|
|
* The GLSL code for evaluating the value. |
|
87
|
|
|
*/ |
|
88
|
|
|
toGLSLFormula(): string { |
|
89
|
1 |
|
const r: string[] = [] |
|
90
|
1 |
|
for (const [exponents, coefficent] of Object.entries(this.coefficents)) { |
|
91
|
5 |
|
const t: string[] = [] |
|
92
|
5 |
|
const f = coefficent.toString() |
|
93
|
5 |
|
if (f !== '1') t.push(f+'.0') |
|
94
|
5 |
|
const dimensions = exponents.split(',') |
|
95
|
5 |
|
for (let i=0; i<dimensions.length; i++) { |
|
96
|
10 |
|
if (dimensions[i] !== '0') { |
|
97
|
6 |
|
t.push(`pow(${this.symbols[i]},${parseInt(dimensions[i], 10)}.0)`) |
|
98
|
|
|
} |
|
99
|
|
|
} |
|
100
|
5 |
|
if (t) r.push(t.join('*')) |
|
101
|
|
|
} |
|
102
|
1 |
|
return r.join('+') |
|
103
|
|
|
} |
|
104
|
|
|
|
|
105
|
|
|
/** |
|
106
|
|
|
* Clone this. |
|
107
|
|
|
*/ |
|
108
|
|
|
clone(): Polyrat { |
|
109
|
1 |
|
return new Polyrat(this.coefficents) |
|
110
|
|
|
} |
|
111
|
|
|
|
|
112
|
|
|
} |
|
113
|
|
|
|
|
114
|
|
|
/** |
|
115
|
|
|
* Parse the string and return it as a Polyrat. |
|
116
|
|
|
*/ |
|
117
|
2 |
|
export const stringToPolyrat = (s: string): Polyrat => { |
|
118
|
1 |
|
return new Polyrat(JSON.parse(s) as Coefficents<Rat>) |
|
119
|
|
|
} |
|
120
|
|
|
|
|
121
|
|
|
export default Polyrat |
|
122
|
|
|
|